package cn.tedu.boot221.controller;

import cn.tedu.boot221.entity.User;
import cn.tedu.boot221.utils.DBUtils;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

@Controller
public class UserController {
    @RequestMapping("/reg")
    @ResponseBody
    public String reg(User user){
        try ( Connection conn = DBUtils.getConn()) {
           String sql = "insert into user values(null,?,?,?,?)";
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getNick());
            ps.setInt(4,user.getAge());
            ps.executeUpdate();
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return "注册成功!";
    }
    @RequestMapping("/login")
    @ResponseBody
    public String login(User user){

        try (Connection conn = DBUtils.getConn()) {
            String sql = "select count(*) from user where username=? and password=?";
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ResultSet rs = ps.executeQuery();
            rs.next();
            if (rs.getInt(1)>0){
                return "登录成功!";
            }
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return "用户名或密码错误!<a href='/login.html'>重新登录</a>";
    }
}
